package cn.evolify.jfinaldemo.base;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.Prop;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.dialect.SqlServerDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.c3p0.C3p0Plugin;
import com.jfinal.plugin.druid.DruidPlugin;

import javax.sql.DataSource;
import java.io.File;

/**
 * Created by wolf on 2017/2/21.
 */
public class ModelGenerator {
    public static DataSource getDataSource() throws ClassNotFoundException {
        System.out.println(PathKit.getWebRootPath());
        Prop p = PropKit.use("config.properties");
        DruidPlugin druidPlugin = new DruidPlugin(p.get("jdbcUrl"), p.get("user"), p.get("password").trim());
        System.out.print(p.get("jdbcUrl"));
        druidPlugin.start();
        return druidPlugin.getDataSource();
    }

    public static void main(String[] args) throws ClassNotFoundException {
        // base model 所使用的包
        String baseModelPackageName = "cn.evolify.jfinaldemo.model.modelbase";
        // base model 文件保存路径
        String baseModelOutputDir = PathKit.getWebRootPath() + "/src/main/java/cn/evolify/jfinaldemo/model/modelbase";

        // model 所使用的包名 (MappingKit 默认使用的包名)
        String modelPackageName = "cn.evolify.jfinaldemo.model";
        // model 文件保存路径 (MappingKit 和 DataDictionary 文件默认保存路径)
        String modelOutputDir = baseModelOutputDir + "/..";

        // 创建生成器
        Generator gernerator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);
        // 设置数据库方言
        gernerator.setDialect(new SqlServerDialect());
        // 添加不需要生成的表名
        gernerator.addExcludedTable("adv");
        // 设置是否在Model 中生成 dao 对象
        gernerator.setGenerateDaoInModel(true);
        // 设置是否生成字典文件
        gernerator.setGenerateDataDictionary(true);
        gernerator.setRemovedTableNamePrefixes("t_");
        // 生成
        gernerator.generate();
    }
}
